import cv2
import numpy as np
from paddleocr import PaddleOCR


def image_text_recognition(image_path, output_dir="ocr_results"):
    """
    使用PaddleOCR识别图片中的文字

    参数:
        video_path: 视频文件路径
        output_dir: 结果保存目录
    """
    # 初始化PaddleOCR，使用中英文模型
    ocr = PaddleOCR(lang="ch")

    # 读取图像
    img = cv2.imread(image_path)
    print(img.shape)

    # 执行OCR识别
    result = ocr.ocr(img)

    # 处理识别结果并保存到文件
    for item in result:  # 遍历结果列表
        texts = item['rec_texts']  # 获取所有识别文本
       # for text in texts:  # 遍历每个识别文本
            # print(text)

image_text_recognition("./test.jpg", "ocr_results")

